import json, re
from pathlib import Path

# [{
#     "query": "生成一个plantuml流程图代码",
#     "response": "@startuml\nstart\n:\"新增系统用户请求\";\nif (\"校验请求合法性\") then (yes)\n :\"查询校验\";\n if (\"数据库中用户不存在\") then (yes)\n  :\"在数据库中新增用户\";\n endif\nendif\n:\"请求结束\";\nstop\n@enduml"
#   },...]
output_json = []
with open('origin.txt') as origin_txt:
    for line in origin_txt:
        line = line.strip()
        ws = re.split(r'\W', line)
        描述 = ' '.join(ws[0:-1])
        类型 = ws[-1]
        print(描述, 类型)
        output_json.append({
            "query": f"输出下面描述的类型，有下面几种类型ILF（内部信息）ELF（外部信息）EI（信息变更相关）EO（统计相关）：{描述}",
            "response": 类型
        })

    # 保存
    Path('data.json').write_text(json.dumps(output_json, ensure_ascii=False))
